package com.zwj.interview.BFS;

/**
 * @Author: Zwj
 * @Date: 2022/2/27 19:41
 */
public class 二叉树中的最近公共节点 {

    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null) {
            return null;
        }
        //这是basecase,只要有一个节点等于root,那么root就是最近公共祖先
        if (root == p || root == q) {
            return root;
        }
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        //如果p,q都在以root为根的树中，那么left和right一定分别是p和q
        if (left != null && right != null) {
            return root;
        }
        //如果p和q都不在以root为根的树中，直接返回null
        if (left == null && right == null) {
            return null;
        }
        //p,q只有一个存在于以root为根的树中，函数返回该节点
        return left == null ? right : left;
    }



    public TreeNode lowestCommon(TreeNode root,TreeNode p,TreeNode q){
        if(root==null){
          return null;
        }
        //有任何一个节点等于root
        if(root==p || root==q){
            return root;
        }
        TreeNode left=lowestCommonAncestor(root.left,p,q);
        TreeNode right=lowestCommonAncestor(root.right,p,q);
        if(left!=null && right!=null){
          return root;
        }
        if(left==null && right==null){
            return null;
        }
        return left==null?right:left;
    }
}